package com.study.hdfs;

import org.apache.hadoop.conf.Configuration;
import org.apache.hadoop.fs.*;
import org.junit.After;
import org.junit.Before;
import org.junit.Test;

import java.io.IOException;
import java.net.URI;
import java.net.URISyntaxException;
import java.util.Arrays;

/**
 * 客户端代码常用套路
 * 1、获取客户端对象
 * 2、执行相关的操作命令
 * 3、关闭资源
 * Hadoop Zookeeper
 * @author weixiaochun
 * @date 2022-5-15 11:32
 */
public class HdfsClient {


    private FileSystem fs;

    @Before
    public void init() throws URISyntaxException, IOException, InterruptedException {
        // 链接的集群nn地址
        URI uri = new URI("hdfs://hadoop102:8020");

        // 配置文件
        Configuration configuration = new Configuration();
        configuration.set("dfs.replication", "2");

        // 用户
        String user = "root";

        // 获取到客户端对象
        fs = FileSystem.get(uri, configuration, user);
    }

    @After
    public void close() throws IOException {
        //关闭资源
        fs.close();
    }

    /**
     * 创建目录
     * @throws URISyntaxException
     * @throws IOException
     * @throws InterruptedException
     */
    @Test
    public void testMkdir() throws URISyntaxException, IOException, InterruptedException {
        // 创建一个文件夹
        fs.mkdirs(new Path("/xiyou/huaguosan1"));
    }

    /**
     * 上传文件
     * 参数优先级
     * hafs-default.xml => hdfs-site.xml => 在项目资源目录下的配置文件 => 代码里面的配置
     */
    @Test
    public void testPut() throws IOException {
        // 参数解读：1、标识删除本地原数据 2、是否允许覆盖 3、源数据路径 4、目的地路径
        fs.copyFromLocalFile(false, true,
                new Path("C:\\Users\\18234\\Documents\\Hadoop学习\\sunwukong.txt"),
                new Path("/xiyou/"));
    }

    /**
     * 文件下载
     */
    @Test
    public void testGet() throws IOException {
        // 参数解读：1、标识删除HDFS原数据 2、源文件路径HDFS 3、目标地址路径Win 4、是否开启本地校验（crc）
        fs.copyToLocalFile(false, new Path("/xiyou/sunwukong.txt"),
                new Path("C:\\Users\\18234\\Documents\\Hadoop学习\\sunwukong.txt"),
                true);
    }

    /**
     * 文件删除
     */
    @Test
    public void testRm() throws IOException {
        // 参数解读：1、删除的路径HDFS 2、是否递归删除
        // 删除文件
        fs.delete(new Path("/jdk-8u161-linux-x64.tar.gz"), false);

        //删除空目录
        fs.delete(new Path("/xiyou"), false);

        //删除非空目录
        fs.delete(new Path("/jinguo"), true);
    }

    /**
     * 文件的更名、移动
     */
    @Test
    public void testMv() throws IOException {
        // 参数解读： 1、源文件路径 2、目标文件的路径
        // 对文件名称的修改
        fs.rename(new Path("/input/word.txt"), new Path("/input/ss.txt"));

        // 文件的移动和更名
        fs.rename(new Path("/input/ss.txt"), new Path("/cls.txt"));

        // 目录的更名
        fs.rename(new Path("/input"), new Path("/output"));
    }

    /**
     * 获取文件详情
     */
    @Test
    public void fileDetail() throws IOException {

        // 获取所有文件信息
        // 参数解读： 1、获取的路径 2、是否递归获取
        RemoteIterator<LocatedFileStatus> listFiles = fs.listFiles(new Path("/"), true);

        // 遍历文件
        while(listFiles.hasNext()){
            LocatedFileStatus fileStatus = listFiles.next();

            // 文件路径
            Path path = fileStatus.getPath();
            System.out.println("======" + path + "======");

            // 文件权限
            System.out.println(fileStatus.getPermission());
            // 文件的所有者
            System.out.println(fileStatus.getOwner());
            // 文件的所属组
            System.out.println(fileStatus.getGroup());
            //文件的大小
            System.out.println(fileStatus.getLen());
            //文件上次修改时间
            System.out.println(fileStatus.getModificationTime());
            //文件的副本数
            System.out.println(fileStatus.getReplication());
            //文件的块大小
            System.out.println(fileStatus.getBlockSize());
            //文件的名称
            System.out.println(fileStatus.getPath().getName());

            // 获取块信息
            BlockLocation[] blockLocations = fileStatus.getBlockLocations();
            System.out.println(Arrays.toString(blockLocations));
        }

    }


    /**
     * 判断是文件夹还是文件
     */
    @Test
    public void isFileOrDir() throws IOException {
        FileStatus[] fileStatuses = fs.listStatus(new Path("/"));
        for (FileStatus status : fileStatuses) {
            if (status.isFile()) {
                System.out.println("文件：" + status.getPath().getName());
            }else{
                System.out.println("目录：" + status.getPath().getName());
            }
        }
    }


}

